package controller;

import java.util.ArrayList;
import java.util.HashMap;

import tools.Statistic;
import tools.TwittParser;

public class Application {

	/**
	 * 
	 * @param arg0 
	 * premier parametre : chemin du fichier d'entrainement
	 * second parametre : chemin du fichier de test
	 * 
	 */
	public static void main(String[] arg0) {
		String path = "data/train.txt";
		if(arg0.length > 0){
			path = arg0[0];
		}
		HashMap<String, ArrayList<String>> dico = TwittParser.parseCorpus(
				path, true, 4, 1);
		for (String s : dico.keySet()) {
			System.out.println(s + " : " + dico.get(s).size());
		}
		Statistic stat = new Statistic(dico);
		stat.getMostFrequents();
		stat.getLeastFrequents();
		
		path = "data/test.txt";
		if(arg0.length > 1){
			path = arg0[1];
		}
		
		ArrayList<String> documents = TwittParser.parseCorpus(path, false, 3);
		System.out.println("\n\nNombre de twitts a trier : "+documents.size());
	}
}
